package com.sxt.shiro.config;

import at.pollux.thymeleaf.shiro.dialect.ShiroDialect;

import org.apache.catalina.security.SecurityUtil;
import org.apache.shiro.SecurityUtils;
import org.apache.shiro.codec.Base64;
import org.apache.shiro.spring.web.ShiroFilterFactoryBean;
import org.apache.shiro.subject.Subject;
import org.apache.shiro.web.mgt.CookieRememberMeManager;
import org.apache.shiro.web.mgt.DefaultWebSecurityManager;
import org.apache.shiro.web.servlet.SimpleCookie;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.beans.factory.annotation.Qualifier;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;

import com.sxt.shiro.service.UserService;

import java.util.LinkedHashMap;
import java.util.Map;

/**
 * @功能描述：shiro核心配置
 * @创建日期: 2019/2/22 18:46
 */
@Configuration
public class ShiroConfig {

	@Autowired
	private UserService userService;

	/**
	 * 创建ShiroFilterFactoryBean shiro过滤bean
	 */
	@Bean
	public ShiroFilterFactoryBean getShiroFilterFactoryBean(
			@Qualifier("securityManager") DefaultWebSecurityManager securityManager) {
		ShiroFilterFactoryBean shiroFilterFactoryBean = new ShiroFilterFactoryBean();

		// 设置安全管理器
		shiroFilterFactoryBean.setSecurityManager(securityManager);

		// 添加Shiro内置过滤器
		/**
		 * Shiro内置过滤器，可以实现权限相关的拦截器 常用的过滤器： anon: 无需认证（登录）可以访问 authc: 必须认证才可以访问 user:
		 * 如果使用rememberMe功能可以直接访问 perms: 该资源必须得到资源权限才可以访问 role: 该资源必须得到角色权限才可以访问
		 * 配置不会被拦截的链接 顺序判断，因为前端模板采用了thymeleaf，这里不能直接使用 ("/static/**",
		 * "anon")来配置匿名访问，必须配置到每个静态目录
		 */

		Map<String, String> filerMap = new LinkedHashMap<String, String>(); // 顺序的map
		// 配置记住我或认证通过可以访问的地址
		filerMap.put("/testThymeleaf", "user");
		// 如果没有拦截，默认会跳转到login.jsp，可以通过setLoginUrl设置登录页面
		// filerMap.put("/add","authc");
		// filerMap.put("/update","authc");
		filerMap.put("/testThymeleaf", "anon");
		filerMap.put("/login", "anon");

		// 授权过滤器
		filerMap.put("/add", "perms[user:add]");
		filerMap.put("/update", "perms[user:update]");
		filerMap.put("/*", "authc");

		// 设置登录的页面，发送toLogin请求
		shiroFilterFactoryBean.setLoginUrl("/toLogin");

		// 设置未授权的页面
		shiroFilterFactoryBean.setUnauthorizedUrl("/noAuth");
		// 设置过滤器
		shiroFilterFactoryBean.setFilterChainDefinitionMap(filerMap);
		return shiroFilterFactoryBean;
	}

	/**
	 * 创建DefaultWebSecurityManager
	 */
	@Bean(name = "securityManager")
	public DefaultWebSecurityManager getDefaultWebSecurityManager(@Qualifier("userRealm") UserRealm userRealm) {
		DefaultWebSecurityManager securityManager = new DefaultWebSecurityManager();
		// 关联realm
		securityManager.setRealm(userRealm);
		securityManager.setRememberMeManager(rememberMeManager());
		return securityManager;
	}

	/**
	 * 创建Realm
	 */
	@Bean(name = "userRealm")
	public UserRealm getRealm() {
		return new UserRealm();
	}

	/**
	 * 配置ShiroDialect，用于thymeleaf和shiro标签配合使用
	 */
	@Bean
	public ShiroDialect getShiroDialect() {
		return new ShiroDialect();
	}

	/**
	 * cookie对象; rememberMeCookie()方法是设置Cookie的生成模版，比如cookie的name，cookie的有效时间等等。
	 * 
	 * @return
	 */
	@Bean
	public SimpleCookie rememberMeCookie() {
		// System.out.println("ShiroConfiguration.rememberMeCookie()");
		// 这个参数是cookie的名称，对应前端的checkbox的name = rememberMe
		SimpleCookie simpleCookie = new SimpleCookie("rememberMe");
		// <!-- 记住我cookie生效时间30天 ,单位秒;-->
		simpleCookie.setMaxAge(259200);
		return simpleCookie;
	}

	/**
	 * cookie管理对象;
	 * rememberMeManager()方法是生成rememberMe管理器，而且要将这个rememberMe管理器设置到securityManager中
	 * 
	 * @return
	 */
	@Bean
	public CookieRememberMeManager rememberMeManager() {
		// System.out.println("ShiroConfiguration.rememberMeManager()");
		CookieRememberMeManager cookieRememberMeManager = new CookieRememberMeManager();
		cookieRememberMeManager.setCookie(rememberMeCookie());
		// rememberMe cookie加密的密钥 建议每个项目都不一样 默认AES算法 密钥长度(128 256 512 位)
		cookieRememberMeManager.setCipherKey(Base64.decode("2AvVhdsgUs0FSA3SDFAdag=="));
		return cookieRememberMeManager;
	}

}
